from selenium import webdriver
from selenium.webdriver.chrome.service import Service
import time

# 1. 手动指定 ChromeDriver 路径
service = Service('D:/devtools/Python/Python312/chromedriver.exe')  # 替换为实际路径

# 2. 创建浏览器实例
chrome = webdriver.Chrome(service=service)
# chrome = webdriver.Chrome()

try:
    # 3. 测试访问网页
    chrome.get("https://www.baidu.com")
    # 4. 截图（确保文件名以 .png 结尾）
    chrome.save_screenshot('google_homepage.png')
    print("截图已保存为: google_homepage.png")
    time.sleep(5)
    chrome.close()  # 关闭当前页
    # input("回车键关闭浏览器...")
    time.sleep(5)
    chrome.quit()
except Exception as e:
    print(f"执行过程中出错: {e}")
# finally:
    # 5. 关闭浏览器
    # chrome.quit()
